batch_size=1
soc_version=Ascend310P3
onnx_path=./pun_model_sim.onnx
output_path=./punctuator_bs${batch_size}

atc --model=${onnx_path} --framework=5 --output=${output_path} --input_format=ND \
--input_shape_range="input_ids:[${batch_size},1~128];token_type_ids:[${batch_size}, 1~128]" --log=error \
--soc_version=${soc_version}